# 导包
import requests
import app
# # 创建封装的类
class IhrmApi:
    def __init__(self):
        self.url_login = app.BASE_URL + "/api/sys/login" # 登陆的URL
        self.url_emp = app.BASE_URL + "/api/sys/user" # 员工模块的URL
    # 封装登陆接口
    def login(self, jsonData):
        # 封装requests发送请求，请求体是动态，从外界接收（形式参数）
        response = requests.post(url=self.url_login,
                                 headers = app.HEADERS,
                                 json=jsonData)
        return response  # 返回响应数据

    # 封装查询员工列表接口
    def query_emp_list(self, headers):
        # 使用requests发送请求，请求头从外界接收
        response = requests.get(url=self.url_emp,
                                params={"page":"1", "size":"10"},
                                headers=headers)
        return response # 返回结果

    # 封装添加员工接口
    def add_emp(self, headers, jsonData):
        return requests.post(url=self.url_emp,
                             headers=headers,
                             json=jsonData)

    # 封装查询员工接口
    def query_emp(self, headers, emp_id):
        query_url = self.url_emp + "/" + emp_id # 拼接查询员工的URL
        print("拼接的查询员工URL为：", query_url)
        return requests.get(url=query_url,headers=headers)


    # 封装修改员工接口
    def modify_emp(self, headers, emp_id, jsonData):
        modify_url = self.url_emp + "/" + emp_id  # 拼接查询员工的URL
        print("拼接的修改员工URL为：", modify_url)
        # 发送请求并返回响应数据
        return requests.put(url=modify_url, headers=headers,
                            json=jsonData)

    # 封装删除员工
    def delete_emp(self, headers, emp_id):
        delete_url = self.url_emp + "/" + emp_id  # 拼接查询员工的URL
        print("拼接的删除员工URL为：", delete_url)
        return requests.delete(url=delete_url, headers=headers)
